#Test file for greedy algorithms with constant cost
use Bio::Phylo::IO;
use SurvivalProbability;
use GreedyAlgorithm;

#Get the tree
$tree_string = "((A:1,B:2)n1:3,C:4)n2:0;";
my $tree = Bio::Phylo::IO->parse(
   '-string' => $tree_string,
   '-format' => 'newick',
)->first;

$sp = SurvivalProbability->new('A','B','C');

my %index = %{$tree->calc_es()};
my %survival = $sp->get_survival_probability_budget_index(2, %index);

use Data::Dumper;
print Dumper(%survival);
